package com.example.tree.bst;

import com.example.tree.TreeNode;

/**
 * 给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于node.val 的值之和。
 *
 *  注意：本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/ 相同
 *  示例 1：
 *  输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
 * 输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
 *                     4
 *            1                 6
 *         0     2          5       7
 *                  3                  8
 *
 *                     30
 *            36                  21
 *         36     35          26       15
 *                   33                     8
 *
 *
 *  示例 2：
 *  输入：root = [0,null,1]
 * 输出：[1,null,1]
 *
 *  示例 3：
 *  输入：root = [1,0,2]
 * 输出：[3,3,2]
 *
 *  示例 4：
 *  输入：root = [3,2,4,1]
 * 输出：[7,9,4,10]
 *
 *      3
 *   2     4
 * 1
 *
 *       7
 *    9     4
 * 10
 */
public class Leetcode538_1038_BstToGst {
    public static void main(String[] args) {
        TreeNode root= new TreeNode(3);
        root.left = new TreeNode(2);
        root.right = new TreeNode(4);
        root.left.left = new TreeNode(1);
        inorderTraversal(root);
        System.out.println("========================");
        TreeNode res = new Solution().convertBST(root);
        inorderTraversal(res);


    }

    static class Solution {
        int sum = 0;// 记录从后向前的累积和

        /**
         * 中序遍历变形(右 -> 根 -> 左)
         * BST的中序遍历(右 -> 根 -> 左)得到的是倒序
         * 使用一个全局变量记录从后向前的累积和
         * @param root
         * @return
         */
        private TreeNode convertBST1(TreeNode root) {
            if (root == null) return null;
            convertBST1(root.right);
            sum += root.val;
            root.val = sum;
            convertBST1(root.left);
            return root;
        }


        public TreeNode convertBST(TreeNode root) {
            return convertBST1(root);
        }
    }

    public static void inorderTraversal(TreeNode root) {
        if (root == null) return;

        inorderTraversal(root.right);
        System.out.println(root.val);
        inorderTraversal(root.left);

    }
}
